package JZ.JZ60;

/**
 * @author corn
 * @create 2020-08-07-19:24
 */

import java.util.ArrayList;
import java.util.LinkedList;

/**
 * JZ60 把二叉树打印成多行
 * 题目描述
 * 从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。
 */
public class TreePrint {

}
class Solution {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> ret = new ArrayList<ArrayList<Integer>>();
        ArrayList<Integer> tmp = new ArrayList<Integer>();
        //存储同一层的结点
        LinkedList<TreeNode> linkCode = new LinkedList<TreeNode>();
        //判空
        if(pRoot == null){
            return ret;
        }
        //添加根结点
        linkCode.add(pRoot);
        //now代表当前层存在的结点个数。next：计算下一层添加的节点数
        int now = 1, next = 0;
        while(!linkCode.isEmpty()) {
            //将当前结点从linkCode提取出来
            TreeNode t = linkCode.remove();
            now--;
            tmp.add(t.val);
            //将当前结点的左右结点连接到后面
            if(t.left != null) {
                linkCode.add(t.left);
                next++;
            }
            if(t.right != null) {
                linkCode.add(t.right);
                next++;
            }

            //当now为0，即当前一层结点全部被提取出来
            if(now == 0) {
                //将当前层数据添加到ret中
                ret.add(new ArrayList<Integer>(tmp));
                //tmp清空
                tmp.clear();
                //得到下一层的结点个数
                now = next;
                //重新再计算下一层结点个数
                next = 0;
            }
        }
        return ret;
    }
}
class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;
    }

}
